# 给定一种 pattern(模式) 和一个字符串 str ，判断 str 是否遵循相同的模式。

# 这里的遵循指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应模式。

# 示例1:

# 输入: pattern = "abba", str = "dog cat cat dog"
# 输出: true
# 示例 2:

# 输入:pattern = "abba", str = "dog cat cat fish"
# 输出: false
# 示例 3:

# 输入: pattern = "aaaa", str = "dog cat cat dog"
# 输出: false
# 示例 4:

# 输入: pattern = "abba", str = "dog dog dog dog"
# 输出: false
# 说明:
# 你可以假设 pattern 只包含小写字母， str 包含了由单个空格分隔的小写字母。


class Solution:
    def wordPattern(self, pattern, str):
        """
        :type pattern: str
        :type str: str
        :rtype: bool
        """
        str_map1, str_map2 = {}, {}
        s=pattern
        t=str.split()
        # s t的长度
        s_list_len = len(s)
        t_list_len = len(t)
        if s_list_len != t_list_len:
            return False

        for i in range(t_list_len):
            if str_map1.get(s[i]) != str_map2.get(t[i]):
                return False
                # 两个字典里对应的值存同一个下标
            str_map1[s[i]] = str_map2[t[i]] = i
        return True

if __name__ == '__main__':
    s = Solution()
    s1 = 'abba'
    s2 = 'dog cat cat fish'
    print(s.wordPattern(s1, s2))